package br.ucb.sigplan.beans;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

public class Cronograma {
	private Map<Atividade, List<Conteudo>> cronograma;
	
	public Cronograma() {
		setCronograma(new TreeMap<Atividade, List<Conteudo>>());
	}
	
	public void addCronograma(Atividade atividade, Conteudo conteudo) {
		if (!getCronograma().containsKey(atividade)) {
			getCronograma().put(atividade, new ArrayList<Conteudo>());
		}
		getCronograma().get(atividade).add(conteudo);
	}
	
	public boolean deleteCronograma(Atividade atividade, Conteudo conteudo) {
		List<Conteudo> conteudos = getCronograma().get(atividade);
		conteudos.remove(conteudo);
		if (conteudos.isEmpty()) {
			return (getCronograma().remove(atividade) != null);
		}
		return true;
	}

	public Map<Atividade, List<Conteudo>> getCronograma() {
		return cronograma;
	}

	public void setCronograma(Map<Atividade, List<Conteudo>> cronograma) {
		this.cronograma = cronograma;
	}
}
